/**
 * Created with IntelliJ IEDA.
 * Description:
 * User:86186
 * Date:2023-11-30
 * Time:22:20
 */

/**
 * 递归:LCR 047. 二叉树剪枝
 * 中等
 * 83
 * 相关企业
 * 给定一个二叉树 根节点 root ，树的每个节点的值要么是 0，要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。
 * 节点 node 的子树为 node 本身，以及所有 node 的后代。
 * 示例 1:
 * 输入: [1,null,0,0,1]
 * 输出: [1,null,0,null,1]
 * 解释:
 * 只有红色节点满足条件“所有不包含 1 的子树”。
 * 右图为返回的答案。
 */
public class pruneTree {
    public TreeNode pruneTree(TreeNode root) {
        if (root == null){
            return null;
        }
        root.left = pruneTree(root.left);
        root.right = pruneTree(root.right);

        if (root.left == null && root.right == null && root.val == 0) {
            root = null;
        }
        return root;
    }

}
